FROM nvcr.io/nvidia/pytorch:21.02-py3

COPY requirements.txt .

RUN pip3 install -r requirements.txt